module.exports = (req, res, next) => {
    res.$success = (data, code = 200) => {
        const _data = {
            code
        }
        if (typeof data === 'object') {
            _data.msg = 'success'
            _data.data = data;
        } else {
            _data.msg = data;
        }
        res.json(_data);
    }

    res.$error = (err, code = 500) => {
        const _data = {
            code
        }
        if (typeof err === 'object') {
            _data.msg = 'error'
            _data.data = JSON.stringify(err);
        } else {
            _data.msg = err;
        }
        res.json(_data);
    }

    next();
}
